import pymysql


class DB:
    def __init__(self):
        self.conn = pymysql.connect(host='localhost', port=3306, user='root', password='123', db='product',
                                    charset='utf8', cursorclass=pymysql.cursors.DictCursor)
        # 获取游标
        self.cousor = self.conn.cursor()

    # 添加修改删除
    def insert(self, sql):
        self.cousor.execute(sql)
        self.conn.commit()

    # 查询单个
    def find(self, sql):
        self.cousor.execute(sql)
        res = self.cousor.fetchone()
        return res

    # 查询所有
    def findall(self, sql):
        self.cousor.execute(sql)
        res = self.cousor.fetchall()
        return res

    # 关闭
    def close(self):
        self.cousor.close()
        self.conn.close()


db1 = DB()
